load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
load("//tensorflow/core/profiler/builds:build_config.bzl", "tf_profiler_copts")

package(
    # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
    default_visibility = ["//tensorflow/core/profiler:internal"],
    licenses = ["notice"],
)

package_group(
    name = "friends",
    includes = [
        "//tensorflow/core/profiler:friends",
    ],
)

# DO NOT ADD NEW DEPENDENCIES TO ANY TARGET IN THIS FILE.
# Instead, use //third_party/xprof/utils.

cc_library(
    name = "diagnostics",
    hdrs = ["diagnostics.h"],
    copts = tf_profiler_copts(),
    visibility = [
        "//perftools/accelerators/xprof/convert:__pkg__",
        "//perftools/gputools/profiler/collector:__pkg__",
    ],
    deps = [
        "@org_xprof//xprof/utils:diagnostics",
    ],
)

cc_library(
    name = "event_span",
    hdrs = ["event_span.h"],
    copts = tf_profiler_copts(),
    visibility = [
        "//perftools/accelerators/xprof/convert:__pkg__",
        "//perftools/gputools/profiler/collector:__pkg__",
    ],
    deps = [
        "@org_xprof//xprof/utils:event_span",
    ],
)

cc_library(
    name = "hardware_type_utils",
    hdrs = ["hardware_type_utils.h"],
    copts = tf_profiler_copts(),
    visibility = ["//perftools/accelerators/xprof/convert:__pkg__"],
    deps = [
        "@org_xprof//xprof/utils:hardware_type_utils",
    ],
)

cc_library(
    name = "math_utils",
    hdrs = ["math_utils.h"],
    visibility = [
        "//perftools/accelerators/xprof/convert:__pkg__",
        "//perftools/accelerators/xprof/xprofilez/integration_tests:__pkg__",
        "//perftools/gputools/profiler/collector:__pkg__",
        "//tensorflow/core/profiler/rpc:__pkg__",
    ],
    deps = [
        "@com_google_absl//absl/base:core_headers",
        "@local_xla//xla/tsl/profiler/utils:math_utils",
    ],
)

cc_library(
    name = "html_utils",
    hdrs = ["html_utils.h"],
    visibility = ["//perftools/gputools/profiler/collector:__pkg__"],
    deps = [
        "@org_xprof//xprof/utils:html_utils",
    ],
)

cc_library(
    name = "op_metrics_db_utils",
    hdrs = ["op_metrics_db_utils.h"],
    copts = tf_profiler_copts(),
    visibility = [
        "//perftools/accelerators/xprof/convert:__pkg__",
        "//perftools/gputools/profiler/collector:__pkg__",
    ],
    deps = [
        "@org_xprof//xprof/utils:op_metrics_db_utils",
    ],
)

cc_library(
    name = "op_utils",
    hdrs = ["op_utils.h"],
    copts = tf_profiler_copts(),
    visibility = ["//perftools/gputools/profiler/collector:__pkg__"],
    deps = [
        "@org_xprof//xprof/utils:op_utils",
    ],
)

cc_library(
    name = "trace_utils",
    hdrs = ["trace_utils.h"],
    copts = tf_profiler_copts(),
    visibility = [
        "//perftools/accelerators/xprof/convert:__pkg__",
        "//perftools/accelerators/xprof/xprofilez/nvidia_gpu:__pkg__",
        "//perftools/gputools/profiler/collector:__pkg__",
    ],
    deps = [
        "@local_xla//xla/tsl/profiler/utils:trace_utils",
    ],
)

cc_library(
    name = "xplane_builder",
    hdrs = ["xplane_builder.h"],
    copts = tf_profiler_copts(),
    visibility = [":friends"],
    deps = [
        "@local_xla//xla/tsl/profiler/utils:xplane_builder",
    ],
)

cc_library(
    name = "xplane_schema",
    hdrs = ["xplane_schema.h"],
    copts = tf_profiler_copts(),
    visibility = [":friends"],
    deps = ["@local_xla//xla/tsl/profiler/utils:xplane_schema"],
)

cc_library(
    name = "xplane_utils",
    hdrs = ["xplane_utils.h"],
    copts = tf_profiler_copts(),
    visibility = [":friends"],
    deps = [
        "@local_xla//xla/tsl/profiler/utils:xplane_utils",
    ],
)

cc_library(
    name = "xplane_test_utils",
    testonly = True,
    hdrs = ["xplane_test_utils.h"],
    copts = tf_profiler_copts(),
    visibility = [
        "//perftools/accelerators/xprof/db:__pkg__",
        "//perftools/gputools/profiler/collector:__pkg__",
    ],
    deps = ["@local_xla//xla/tsl/profiler/utils:xplane_test_utils"],
)

cc_library(
    name = "xplane_visitor",
    hdrs = ["xplane_visitor.h"],
    copts = tf_profiler_copts(),
    visibility = [":friends"],
    deps = [
        "@local_xla//xla/tsl/profiler/utils:xplane_visitor",
    ],
)

cc_library(
    name = "cost_utils",
    hdrs = ["cost_utils.h"],
    copts = tf_profiler_copts(),
    visibility = ["//visibility:private"],
    deps = [
        "@org_xprof//xprof/utils:cost_utils",
    ],
)

cc_library(
    name = "derived_timeline",
    hdrs = ["derived_timeline.h"],
    copts = tf_profiler_copts(),
    visibility = [
        "//perftools/accelerators/xprof/convert:__pkg__",
        "//perftools/accelerators/xprof/xplane:__pkg__",
        "//platforms/darwinn/tools/xprof_trace:__pkg__",
    ],
    deps = [
        "@org_xprof//xprof/utils:derived_timeline",
    ],
)

cc_library(
    name = "kernel_stats_utils",
    hdrs = ["kernel_stats_utils.h"],
    copts = tf_profiler_copts(),
    visibility = [
        "//perftools/accelerators/xprof/convert:__pkg__",
        "//perftools/gputools/profiler/collector:__pkg__",
    ],
    deps = ["@org_xprof//xprof/utils:kernel_stats_utils"],
)

cc_library(
    name = "tfstreamz_utils",
    hdrs = ["tfstreamz_utils.h"],
    copts = tf_profiler_copts(),
    visibility = ["//perftools/accelerators/xprof/xprofilez/cpu:__pkg__"],
    deps = [
        "@org_xprof//xprof/utils:tfstreamz_utils",
    ],
)

cc_library(
    name = "step_intersection",
    hdrs = ["step_intersection.h"],
    copts = tf_profiler_copts(),
    visibility = ["//perftools/accelerators/xprof/convert:__pkg__"],
    deps = [
        "@org_xprof//xprof/utils:step_intersection",
    ],
)

cc_library(
    name = "device_caps_utils",
    hdrs = ["device_caps_utils.h"],
    copts = tf_profiler_copts(),
    visibility = [
        "//perftools/accelerators/xprof/xplane:__pkg__",
        "//platforms/xla/tools:__pkg__",
    ],
    deps = ["@org_xprof//xprof/utils:device_caps_utils"],
)

cc_library(
    name = "gpu_event_stats",
    hdrs = ["gpu_event_stats.h"],
    copts = tf_profiler_copts(),
    visibility = [
        "//perftools/accelerators/xprof/convert:__pkg__",
        "//perftools/gputools/profiler/collector:__pkg__",
        "//tensorflow/core/profiler/convert:__pkg__",
    ],
    deps = [
        "@org_xprof//xprof/utils:gpu_event_stats",
    ],
)

cc_library(
    name = "hlo_proto_map",
    hdrs = ["hlo_proto_map.h"],
    visibility = [
        "//perftools/accelerators/xprof/convert:__pkg__",
        "//perftools/accelerators/xprof/xplane:__pkg__",
        "//perftools/accelerators/xprof/xprofilez/integration_tests:__pkg__",
        "//perftools/gputools/profiler/collector:__pkg__",
    ],
    deps = [
        "@org_xprof//xprof/utils:hlo_proto_map",
    ],
)

cc_library(
    name = "hlo_proto_to_module",
    hdrs = ["hlo_proto_to_module.h"],
    visibility = [
        "//perftools/accelerators/xprof/convert:__pkg__",
        "//perftools/gputools/profiler/collector:__pkg__",
    ],
    deps = [
        "@org_xprof//xprof/utils:hlo_proto_to_module",
    ],
)

cc_library(
    name = "hlo_module_map",
    hdrs = ["hlo_module_map.h"],
    visibility = ["//perftools/accelerators/xprof/convert:__pkg__"],
    deps = [
        "@org_xprof//xprof/utils:hlo_module_map",
    ],
)

cc_library(
    name = "hlo_module_utils",
    hdrs = ["hlo_module_utils.h"],
    visibility = [
        "//perftools/accelerators/xprof/convert:__pkg__",
        "//perftools/gputools/profiler/collector:__pkg__",
        "//tensorflow/compiler/mlir/lite/experimental/google/tooling/hlo_adapter:__pkg__",
    ],
    deps = [
        "@org_xprof//xprof/utils:hlo_module_utils",
    ],
)

cc_library(
    name = "xprof_gpu_cost_analysis",
    hdrs = ["xprof_gpu_cost_analysis.h"],
    visibility = ["//visibility:private"],
    deps = [
        "@org_xprof//xprof/utils:xprof_gpu_cost_analysis",
    ],
)

cc_library(
    name = "tpu_step_breakdown_utils",
    hdrs = ["tpu_step_breakdown_utils.h"],
    visibility = ["//perftools/accelerators/xprof/convert:__pkg__"],
    deps = [
        "@org_xprof//xprof/utils:tpu_step_breakdown_utils",
    ],
)

cc_library(
    name = "tpu_step_details_utils",
    hdrs = ["tpu_step_details_utils.h"],
    visibility = ["//perftools/gputools/profiler/collector:__pkg__"],
    deps = [
        "@org_xprof//xprof/utils:tpu_step_details_utils",
    ],
)
